package com.example.heartprotectionserver;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;

/**
 * 代码生成器
 */
public class Generator {
    public static void main(String[] args) {
        //核心代码
        AutoGenerator autoGenerator = new AutoGenerator();
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        //连接数据库
        dataSourceConfig.setDriverName("com.mysql.cj.jdbc.Driver");
        dataSourceConfig.setUrl("jdbc:mysql://localhost:3306/hp?allowPublicKeyRetrieval=true&useSSL=false");
        dataSourceConfig.setUsername("root");
        dataSourceConfig.setPassword("wen200389");
        autoGenerator.setDataSource(dataSourceConfig);

        //全局配置
        GlobalConfig globalConfig = new GlobalConfig();
        globalConfig.setOutputDir(System.getProperty("user.dir")+"/src/main/java");//设置生成代码存放的位置
        globalConfig.setOpen(false);//设置生成完毕后是否打开生成代码所在的目录
        globalConfig.setAuthor("张力文");//设置作者
        globalConfig.setFileOverride(true);//设置是否覆盖原始生成的文件
//        globalConfig.setMapperName("%sDao");//设置数据层接口名，%s为占位符，指代模块名称
        globalConfig.setIdType(IdType.AUTO);//设置Id生成策略
        autoGenerator.setGlobalConfig(globalConfig);

        //包设置
        PackageConfig packageConfig = new PackageConfig();
        packageConfig.setParent("com.example.heartprotectionserver");//设置父包名
        packageConfig.setEntity("model");//设置实体类包名
//        packageConfig.setMapper("dao");//设置数据层包名
        autoGenerator.setPackageInfo(packageConfig);

        //策略设置
        StrategyConfig strategyConfig = new StrategyConfig();
//        strategyConfig.setInclude("tb_admin");//设置要生成哪些表
        strategyConfig.setTablePrefix("tb_");//设置数据库表的前缀名称
        strategyConfig.setRestControllerStyle(true);//设置是否启用Rest风格
        //根据情况进行设置
//        strategyConfig.setVersionFieldName("version");//设置乐观锁字段
//        strategyConfig.setLogicDeleteFieldName("deleted");//设置逻辑删除字段
        strategyConfig.setEntityLombokModel(true);//设置是否启用lombok
        autoGenerator.setStrategy(strategyConfig);
        //开始生成，默认生成到D://
        autoGenerator.execute();
    }
}
